from crewai import Agent, Crew, Task, Process
import models


# 定义 Agent 1：数据分析师
agent_one = Agent(
    llm=models.get_model(),
    role="数据分析师",
    goal="分析市场中的数据趋势",
    backstory="一位具有经济学背景的资深数据分析师",
    verbose=True,

)

# 定义 Agent 2：市场研究员
agent_two = Agent(
    llm=models.get_model(),
    role="市场研究员",
    goal="收集有关市场动态的信息",
    backstory="一位注重细节、勤奋的研究员",
    verbose=True,

)

# 定义任务 1：收集市场数据并识别趋势
task_one = Task(
    description="收集最近的市场数据并识别趋势。",
    expected_output="一份总结市场关键趋势的报告。",
    agent=agent_one
)

# 定义任务 2：研究影响市场动态的因素
task_two = Task(
    description="研究影响市场动态的因素。",
    expected_output="一份对影响市场因素的分析报告。",
    agent=agent_two,
)

# 组建 Crew 并按顺序执行任务
crew = Crew(
    agents=[agent_one, agent_two],
    tasks=[task_one, task_two]
)


result = crew.kickoff()
print(result)
